package cn.ch.system.service.impl;

import cn.ch.basic.service.impl.BaseServiceImpl;
import cn.ch.system.domain.Menu;
import cn.ch.system.mapper.MenuMapper;
import cn.ch.system.service.IMenuService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author Why
 * @since 2022-02-20
 */
@Service
public class MenuServiceImpl extends BaseServiceImpl<Menu> implements IMenuService {
    @Autowired
    private MenuMapper menuMapper;

    @Transactional
    public void save(Menu menu) {
        if (Objects.nonNull(menu.getParent())){//非顶级部门
            menu.setParentId(menu.getParent().getId());
        }
        super.add(menu);
    }

    /**
     * 查询无限级树
     * @return list集合 只有顶级菜单的数据
     */
    @Override
    public List<Menu> queryTree() {
        //获取所有的菜单信息
        List<Menu> all = menuMapper.loadAll();
        //创建需要返回的顶层集合
        List<Menu> top = new ArrayList<>();
        //将所有的菜单转储为map集合
        Map<Long, Menu> menuMap = all.stream().collect(Collectors.toMap(Menu::getId, o -> o));
        for (Menu d: all) {
            //找出树的顶层
            if (d.getParentId()==null){
                top.add(d);
            }else {
                //找到自己的父级  并将自己加进去
                Menu parent = menuMap.get(d.getParentId());
                if (parent!=null)
                    parent.getChildren().add(d);
            }
        }
        return top;//返回最顶层
    }

/*    @Transactional
    public void add(Menu menu) {
        menuMapper.save(menu);//返回自增id
        this.update(menu);//抽取
    }

    @Override
    @Transactional
    public void update(Menu menu) {
        //这里不仅需要保存数据，还需要保存ditPath 后期用来做上级部门回显
        String dirPath = "";
        if(menu.getParent() == null){
            //1.如果新增的是定顶级 直接保存自己的自增id  /自己的id
            dirPath = "/"+menu.getId();
        }else{
            //2.如果不是顶级  找出父亲的dirPath + /自己的id
            Menu parent = menuMapper.loadById(menu.getParent().getId());
            dirPath = parent.getComponent() + "/"+menu.getId();
        }
        menu.setComponent(dirPath);
        menuMapper.update(menu);
    }

    @Override
    public List<Menu> queryTree() {
//        *  方案四：必须掌握
//        *    1.一次性查询出所有数据，放到内存 all
        List<Menu> allDepts = menuMapper.loadAll();
        //为了下面内层嵌套循环，查询父亲方便，我们这里把所有数据，抽成map<部门ID，部门对象>
        *//*Map<Long, Menu> allDeptMap =
                allDepts.stream().collect(Collectors.toMap(Menu::getId, dept -> dept));*//*
        HashMap<Long, Menu> allDeptMap = new HashMap<>();
        for (Menu tmp : allDepts) {
            allDeptMap.put(tmp.getId(),tmp);
        }

        List<Menu> firstDepts = new ArrayList<>();
        for (Menu deptTmp : allDepts) {
            if(deptTmp.getParent_id() == null){//代表是顶级
//        *    2.从所有数据中找出第一层级 firstMenu   我们需要返回的数据
                firstDepts.add(deptTmp);
            }else{
                // 可能是第二、第三、第四......
//        *    3.只要不是第一层级的，找自己的上级，并将自己放入上级的children中
                Menu parent = allDeptMap.get(deptTmp.getParent_id());
                if(parent != null){
                    parent.getChildren().add(deptTmp);
                }

                *//*for(Menu parent : allDepts){
                    if(deptTmp.getParent().getId() == parent.getId()){
                        //把我加到你的儿子里面
                        parent.getChildren().add(deptTmp);
                    }
                    continue;
                }*//*
            }
        }
        return firstDepts;
    }*/
}
